네이버 부스트캠프 [PyTorch] Transfer Learning 1️⃣ 모델 architecture 저장 2️⃣ 파라미터 저장 0) 모델 파라미터 출력 : model.state_dict() 1) 모델 파라미터 저장 및 불러오기 : model.load_state_dict() 2) 모델 아키텍처 + 파라미터 저장 및 불러오기 (pickle 형태로 저장) : torch.load() torch.load 자체로 객체가 생성되어 한문장으로 모델과 파라미터 둘다 불러... 네이버 부스트캠프PyTorchPyTorch
[PyTorch] Transfer Learning 1️⃣ 모델 architecture 저장 2️⃣ 파라미터 저장 0) 모델 파라미터 출력 : model.state_dict() 1) 모델 파라미터 저장 및 불러오기 : model.load_state_dict() 2) 모델 아키텍처 + 파라미터 저장 및 불러오기 (pickle 형태로 저장) : torch.load() torch.load 자체로 객체가 생성되어 한문장으로 모델과 파라미터 둘다 불러... 네이버 부스트캠프PyTorchPyTorch